home *** CD-ROM | disk | FTP | other *** search
/ Cream of the Crop 11 / Cream of the Crop 11-1.iso / compress / gnucpio.zip / XMALLOC.C < prev    next >
C/C++ Source or Header  |  1995-06-24  |  2KB  |  104 lines
  1. /* xmalloc.c -- malloc with out of memory checking
  2.    Copyright (C) 1990, 91, 92, 93, 94 Free Software Foundation, Inc.
  3.  
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 2, or (at your option)
  7.    any later version.
  8.  
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.  
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17.  
  18. #ifdef HAVE_CONFIG_H
  19. #include <config.h>
  20. #endif
  21.  
  22. #if __STDC__
  23. #define VOID void
  24. #else
  25. #define VOID char
  26. #endif
  27.  
  28. #include <sys/types.h>
  29.  
  30. #if STDC_HEADERS
  31. #include <stdlib.h>
  32. #else
  33. VOID *malloc ();
  34. VOID *realloc ();
  35. void free ();
  36. #endif
  37.  
  38. /* This is for other GNU distributions with internationalized messages.
  39.    The GNU C Library itself does not yet support such messages.  */
  40. #if HAVE_LIBINTL_H
  41. # include <libintl.h>
  42. #else
  43. # define gettext(msgid) (msgid)
  44. #endif
  45.  
  46. #ifndef EXIT_FAILURE
  47. #define EXIT_FAILURE 1
  48. #endif
  49.  
  50. /* Exit value when the requested amount of memory is not available.
  51.    The caller may set it to some other value.  */
  52. int xmalloc_exit_failure = EXIT_FAILURE;
  53.  
  54. #if __STDC__ && (HAVE_VPRINTF || HAVE_DOPRNT)
  55. void error (int, int, const char *, ...);
  56. #else
  57. void error ();
  58. #endif
  59.  
  60. static VOID *
  61. fixup_null_alloc (n)
  62.      size_t n;
  63. {
  64.   VOID *p;
  65.  
  66.   p = 0;
  67.   if (n == 0)
  68.     p = malloc ((size_t) 1);
  69.   if (p == 0)
  70.     error (xmalloc_exit_failure, 0, gettext ("Memory exhausted"));
  71.   return p;
  72. }
  73.  
  74. /* Allocate N bytes of memory dynamically, with error checking.  */
  75.  
  76. VOID *
  77. xmalloc (n)
  78.      size_t n;
  79. {
  80.   VOID *p;
  81.  
  82.   p = malloc (n);
  83.   if (p == 0)
  84.     p = fixup_null_alloc (n);
  85.   return p;
  86. }
  87.  
  88. /* Change the size of an allocated block of memory P to N bytes,
  89.    with error checking.
  90.    If P is NULL, run xmalloc.  */
  91.  
  92. VOID *
  93. xrealloc (p, n)
  94.      VOID *p;
  95.      size_t n;
  96. {
  97.   if (p == 0)
  98.     return xmalloc (n);
  99.   p = realloc (p, n);
  100.   if (p == 0)
  101.     p = fixup_null_alloc (n);
  102.   return p;
  103. }
  104.